import pandas as pd
from pandasai import Agent
from pandasai.llm.openai import OpenAI

# 读取 CSV 文件，使用原始字符串避免转义问题
df = pd.read_csv("D:\宁德时代股票.csv")
# 将日期 列转换为date时间格式
df['日期'] = pd.to_datetime(df['日期'])
# 设置 日期 列为索引
df.set_index('日期', inplace=True)

# 配置 Deepseek 模型
OpenAI._supported_chat_models.append('deepseek-chat')
llm_openai = OpenAI(
    api_base='https://api.deepseek.com/v1',
    api_token='sk-d57a50bd4d19428583093ee15d3e4510',  # 替换为实际的 API 令牌
    model='deepseek-chat'
)

# 创建 Agent（强制纯文本输出，禁用图表和缓存）
agent = Agent(
    dfs=[df], 
    config={
        "llm": llm_openai,
        "save_charts": False,    # 禁用图表生成
        "enable_cache": False,   # 禁用缓存
        "response_style": "text" # 强制纯文本响应
    }
)

# 对股价进行数据分析，为投资决策提供支持
investment_analysis_prompt = "基于股价数据评估潜在风险和市场机会，为投资决策提供建议"
print("正在生成纯文本股价分析报告...")

